package com.geline.web.interceptor;

import cn.hutool.extra.servlet.ServletUtil;
import cn.hutool.http.ContentType;
import cn.hutool.json.JSONUtil;
import com.geline.cloud.domain.R;
import com.geline.web.annotation.RepeatSubmit;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.HandlerInterceptor;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.lang.reflect.Method;

/**
 * 防止重复提交拦截器
 * 使用方式：
     @Override
     public void addInterceptors(InterceptorRegistry registry) {
         registry.addInterceptor(new RepeatSubmitInterceptor()).addPathPatterns("/**");
     }
 * @RepeatSubmit
 * @author ruoyi
 */
@Slf4j
public abstract class RepeatSubmitInterceptor implements HandlerInterceptor {

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        if (handler instanceof HandlerMethod) {
            HandlerMethod handlerMethod = (HandlerMethod) handler;
            Method method = handlerMethod.getMethod();
            RepeatSubmit annotation = method.getAnnotation(RepeatSubmit.class);
            if (annotation != null && this.isRepeatSubmit(request)) {
                response.setCharacterEncoding("utf-8");
                ServletUtil.write(response, JSONUtil.toJsonStr(R.fail("不允许重复提交，请稍后再试")), ContentType.JSON.toString());
                return false;
            }
        }
        return true;
    }

    /**
     * 验证是否重复提交由子类实现具体的防重复提交的规则
     * @param request
     * @return
     * @throws Exception
     */
    public abstract boolean isRepeatSubmit(HttpServletRequest request);
}
